import cv2
import pytesseract

def read_code():
    # 读取图像
    image_path = 'code.png'  # 请替换为你的验证码图片路径
    image = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)

    # 预处理图像
    # 1. 转换为灰度图（如果还没有）
    # 2. 二值化处理
    _, binary_image = cv2.threshold(image, 127, 255, cv2.THRESH_BINARY_INV)

    # 3. 去噪（可选，根据实际情况调整）
    binary_image = cv2.medianBlur(binary_image, 3)

    # 4. 显示预处理后的图像（可选）
    cv2.imshow('Binary Image', binary_image)


    # 使用 Tesseract 进行 OCR 识别
    captcha_text = pytesseract.image_to_string(binary_image, config='--psm 7 digits')

    # 输出识别结果
    print("识别的验证码为:", captcha_text.strip())
    return captcha_text.strip()




# cv2.waitKey(0)
# cv2.destroyAllWindows()